#ifndef AES_H
#define AES_H

    #include "rijndael.h"

    #include <string>
    #define KEYBITS       256

    class AES
    {
        private:
            unsigned long rk[RKLENGTH(KEYBITS)];
            unsigned char key[KEYLENGTH(KEYBITS)];

        public:
            void setkey(std::string);
            unsigned int encrypt(const char *, char *, unsigned int,
                unsigned int);
            unsigned int decrypt(const char *, char *, unsigned int,
                unsigned int);
    };

#endif
